/*
 * MIT License
 *
 * Copyright (c) 2020 wen.gu <454727014@qq.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

/***************************************************************************
* Name: av_decoder.h
*
* Purpose: decoder component implementation with libav.
*
* Developer:
*   wen.gu , 2018-12-28
*
* TODO:
*
***************************************************************************/
#ifndef __AV_DECODER_H__
#define __AV_DECODER_H__

/******************************************************************************
**    INCLUDES
******************************************************************************/


#include "mp_decoder.h"
extern "C"
{
#include "libavcodec/avcodec.h"
}

/******************************************************************************
**    MACROS
******************************************************************************/


/******************************************************************************
**    TYPE DEFINITIONS
******************************************************************************/

/******************************************************************************
**    CLASSES/FUNCTIONS DEFINITIONS
******************************************************************************/

class AvDecoder : public MPDecoder
{
public:
    AvDecoder();
    ~AvDecoder();


public:
    virtual MPError reset() override; /** reset decoder state to new decode */
    virtual MPTrackInfo* getFormat() override; /**get decoder output information*/
    virtual MPError sendPacket(MPBuffer* pkt) override;
    virtual MPError receiveFrame(MPBuffer** frame) override;

public:
    MPError initialize(const MPTrackInfo& info);

protected:
    AVCodecID convert2avId(eMediaFormat fmt);
    eMediaFormat convert2Fmt(AVPixelFormat id);
    eMediaFormat convert2Fmt(AVSampleFormat id);
    MPError processAudioOutput(AVFrame& avf, MPBuffer** frame);
    MPError processVideoOutput(AVFrame& avf, MPBuffer** frame);
    void audioPlanar2Interleaved(uint8_t* dst, AVFrame& frame);
    uint32_t getFrameSize(AVFrame& avf);
private:
    const AVCodec *mCodec = nullptr;
    AVCodecContext *mCtx = nullptr;
    MPTrackInfo mInfo;

};


#endif /** !__AV_DECODER_H__ */